import requests
import unittest


class TestIhrmLogin(unittest.TestCase):
    def test_login_ok(self):
        res = requests.post(url="http://ihrm.itheima.net/prod-api/sys/login", json={
            "mobile": "13800000002",
            "password": "123456"
        })
        print(res.json())
        # 断言 响应状态码为200
        self.assertEqual(200, res.status_code)
        self.assertEqual(True, res.json().get("success"))
        self.assertEqual(10000, res.json().get("code"))
        self.assertIn("成功", res.json().get("message"))

    def test_login_phone_not_exist(self):
        res = requests.post(url="http://ihrm.itheima.net/prod-api/sys/login", json={
            "mobile": "13800000009",
            "password": "123456"
        })
        print(res.json())
        # 断言 响应状态码为200
        self.assertEqual(200, res.status_code)
        self.assertEqual(False, res.json().get("success"))
        self.assertEqual(10000, res.json().get("code"))
        self.assertIn("用户名或密码错误", res.json().get("message"))

    def test_login_pwd_failed(self):
        res = requests.post(url="http://ihrm.itheima.net/prod-api/sys/login", json={
            "mobile": "13800000002",
            "password": "12345678"
        })
        print(res.json())
        # 断言 响应状态码为200
        self.assertEqual(200, res.status_code)
        self.assertEqual(False, res.json().get("success"))
        self.assertEqual(10000, res.json().get("code"))
        self.assertIn("用户名或密码错误", res.json().get("message"))
